#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <assert.h>
#include <unistd.h>
#include <errno.h>
#include <string.h>
#include <stdio.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <fcntl.h>
#include <stdlib.h>
#include<sys/wait.h>
void handle_signal(int sig)
{
    printf("service recv a signal %d\n", sig);
    exit(1);
}
int main()
{
    int listenfd = socket(PF_INET, SOCK_STREAM, 0);
    if (listen < 0)
    {
        perror("socket:");
        exit(1);
    }

    struct sockaddr_in servaddr;
    memset(&servaddr, 0, sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    servaddr.sin_port = htons(5188);
    //设置地址可以重复利用
    int on = 1;
    if ((setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on))) < 0)
    {
        perror("setsockopt ");
        exit(1);
    }
    int ret = bind(listenfd, (struct sockaddr *)&servaddr, sizeof(servaddr));
    if (ret < 0)
    {
        perror("bind");
        exit(1);
    }

    ret = listen(listenfd, SOMAXCONN);
    if (ret < 0)
    {
        perror("listen");
        exit(1);
    }
    struct sockaddr_in peeraddr;
    socklen_t peeraddrlen = sizeof(peeraddr);
    int connfd = accept(listenfd, (struct sockaddr *)&peeraddr, &peeraddrlen);
    if (connfd < 0)
    {
        perror("accept");
        exit(1);
    }

    printf("ip:%s port:%d\n", inet_ntoa(peeraddr.sin_addr), ntohs(peeraddr.sin_port));
    //一个进程处理收数据，一个处理发数据
    pid_t pid = fork();
    if (pid < 0)
    {
        perror("fork");
        exit(1);
    }
    else if (pid == 0)
    {
        signal(SIGUSR1,handle_signal);
        char sendbuf[1024] = {0};
        while(fgets(sendbuf,sizeof(sendbuf),stdin) != NULL) //服务器从键盘获取数据
        {
            write(connfd, sendbuf,sizeof(sendbuf)); //服务器发送数据给客户端
            memset(sendbuf,0,sizeof(sendbuf));
        }
        close(connfd);
        exit(1);      
    }
    else
    {
        char recvbuf[1024];
        while (1)
        {
            memset(recvbuf, 0, sizeof(recvbuf));
            int sz = read(connfd, recvbuf, sizeof(recvbuf)); //服务器接收客户端发来的数据
            if (sz < 0)
            {
                perror("read");
                exit(1);
            }
            else if (sz == 0)
            {
                printf("peer close\n");
                break;
            }
            fputs(recvbuf, stdout);     //服务器打印客户端发来的数据
        }
        close(connfd);
        close(listenfd);
        kill(pid,SIGUSR1);
        exit(1);
    }
    return 0;
}